#!/usr/bin/python3
# _*_ coding: utf-8 _*_
#
# Copyright (C) 2025 - 2025 anjiu, Inc. All Rights Reserved 
#
# @Time    : 2025/9/18 10:43
# @Author  : anjiu
# @Email   : basui6996@gmail.com
# @File    : fix_project_structure.py
# @IDE     : PyCharm
# @Description    : 修复项目结构脚本

import os
import shutil


def create_src_structure():
    """
    创建标准的 src 项目结构

    :return:
    """
    # 创建 src 目录
    if not os.path.exists("src"):
        os.makedirs("src")
        print("创建 src 目录")

    # 移动 data_chunker 包到 src 目录
    if os.path.exists("data_chunker") and not os.path.exists("src/data_chunker"):
        shutil.move("data_chunker", "src/data_chunker")
        print("移动 data_chunker 到 src 目录")

    # 确保必要的目录存在
    for dir_name in ["tests", "examples", "docs"]:
        if not os.path.exists(dir_name):
            os.makedirs(dir_name)
            print(f"创建 {dir_name} 目录")

    print("项目结构修复完成!")


if __name__ == "__main__":
    print("=== 修复项目结构 ===")
    create_src_structure()
